package com.xtoneict.util.jwt;

import io.jsonwebtoken.*;
import io.jsonwebtoken.security.Keys;
import org.springframework.util.StringUtils;

import java.util.Date;


/**
 * jwt工具类
 */
public class JwtHelper {

//    //TODO 修改token有效时间
//    private static long tokenExpiration = 365 * 24 * 60 * 60 * 1000;
//    private static String tokenSignKey = "123456";
//
//    /**
//     * 根据用户id和名称生成token字符串
//     * @param userId
//     * @param username
//     * @return
//     */
//    public static String createToken(int userId, String username) {
//        String token = Jwts.builder()
//                .setSubject("AUTH-USER")
//                //有效时长
//                .setExpiration(new Date(System.currentTimeMillis() + tokenExpiration))
//                //主体部分
//                .claim("userId", userId)
//                .claim("username", username)
//                //签名部分
//                .signWith(SignatureAlgorithm.HS512, tokenSignKey)
//                .compressWith(CompressionCodecs.GZIP)
//                .compact();
//        return token;
//    }

    //TODO 修改token有效时间
    private static long tokenExpiration = 365 * 24 * 60 * 60 * 1000;
    //    private static String tokenSignKey = "123456";
    public static final byte[] tokenSignKey = Keys.secretKeyFor(SignatureAlgorithm.HS512).getEncoded();
    /**
     * 根据用户id和名称生成token字符串
     * @param userId
     * @param username
     * @return
     */
    public static String createToken(int userId, String username) {
        String token = Jwts.builder()
                .setSubject("AUTH-USER")
                //有效时长
                .setExpiration(new Date(System.currentTimeMillis() + tokenExpiration))
                //主体部分
                .claim("userId", userId)
                .claim("username", username)
                //签名部分
//                .signWith(SignatureAlgorithm.HS512, tokenSignKey)
                .signWith(Keys.hmacShaKeyFor(tokenSignKey), SignatureAlgorithm.HS512)
                .compressWith(CompressionCodecs.GZIP)
                .compact();
        return token;
    }

    /**
     * 解析token
     * @param
     * @return
     */
//    public static Claims parseToken(String token) {
//        Claims claims = Jwts.parserBuilder()
//                .setSigningKey(Keys.hmacShaKeyFor(tokenSignKey)) // 使用与生成时相同的密钥
//                .build()
//                .parseClaimsJws(token)
//                .getBody();
//        return claims;
//    }
    public static Claims parseToken(String jwt){
        if(!StringUtils.hasLength(jwt)){
            return null;
        }

        JwtParserBuilder jwtParserBuilder = Jwts.
                parserBuilder()
                .setSigningKey(tokenSignKey);
        Claims claims = null;
        try {
            claims = jwtParserBuilder.build().parseClaimsJwt(jwt).getBody();
        }catch (Exception e){

        }
        return claims;
    }

    /**
     * 从token中获取用户id
     * @param token
     * @return
     */
    public static Long getUserId(String token) {
        try {
            if (StringUtils.isEmpty(token)) {
                return null;
            }

            Jws<Claims> claimsJws = Jwts.parser().setSigningKey(tokenSignKey).parseClaimsJws(token);
            Claims claims = claimsJws.getBody();
            Integer userId = (Integer) claims.get("userId");
            return userId.longValue();
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * 从token中获取用户名称
     * @param token
     * @return
     */
    public static String getUsername(String token) {
        try {
            if (StringUtils.isEmpty(token)) {
                return "";
            }

            Jws<Claims> claimsJws = Jwts.parser().setSigningKey(tokenSignKey).parseClaimsJws(token);
            Claims claims = claimsJws.getBody();
            return (String) claims.get("username");
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

}
